from django.db import models
from django.core.validators import MinValueValidator, MaxValueValidator


# Create your models here.


class WatermarkTemplate(models.Model):
    """文件水印模板"""
    create_time = models.DateTimeField(auto_now_add=True)       # 创建时间
    name = models.CharField(max_length=50, unique=True)         # 模板名称
    content = models.CharField(max_length=100)                  # 水印文字
    rotate_angle = models.IntegerField(validators=[             # 旋转角度
        MinValueValidator(-180),
        MaxValueValidator(180),
    ])
    opacity = models.PositiveIntegerField(                      # 不透明度
        default=100, validators=[MaxValueValidator(100)]
    )


class FileManagement(models.Model):
    """文件管理模型"""
    create_time = models.DateTimeField(auto_now_add=True)       # 创建时间
    file_md5 = models.CharField(max_length=50, default="")      # 文件md5值
    name = models.CharField(max_length=50)                      # 文件名称
    level = models.SmallIntegerField(default=0)                 # 敏感等级，1表示低危，2表示中危，3表示高危
    file_type = models.SmallIntegerField(default=0)             # 文件类型，1为文件匹配，2为关键字匹配，3为相似度匹配
    sensitive_word = models.CharField(max_length=50, default="")    # 敏感词，当且仅当file_type为2时才有值
    status = models.SmallIntegerField(default=1)                # 表示状态，1为有效，0为已删除
